from langchain.prompts import PromptTemplate
from langchain_core.runnables import RunnableSequence
from langchain_openai import ChatOpenAI

llm = ChatOpenAI(model_name="gpt-4o-mini", temperature=0)
prompt = PromptTemplate.from_template(
    "你是一名计算机学科的老师，需要根据学生提出的问题，给出完整的解答结果。学生的提问是： {student_msg}。")
chain = RunnableSequence(llm, prompt)
student_msg = "列出两种常见的数据结构，并解释它们的用途"
res = chain.invoke(student_msg)
print(res)
